package com.wave.wavesystem.ai.advisor.factory;

import org.springframework.ai.chat.prompt.PromptTemplate;
import org.springframework.ai.rag.generation.augmentation.ContextualQueryAugmenter;

/**
 * 创建上下文查询增强器的 工厂，
 */
public class PostContextualQueryAugmenterFactory {

    public static ContextualQueryAugmenter create() {
        PromptTemplate emptyPrompt = new PromptTemplate("""
                你应该输出下面的内容：
                抱歉，我只能回答恋爱相关的问题，有问题可以联系
                站长 小牛同学
                """);

        return ContextualQueryAugmenter.
                builder()
                .allowEmptyContext(false)
                .emptyContextPromptTemplate(emptyPrompt)
                .build();
    }
}
